/*
请你给一个停车场设计一个停车系统。停车场总共有三种不同大小的车位：大，中和小，每种尺寸分别有固定数目的车位。

请你实现 ParkingSystem 类：

ParkingSystem(int big, int medium, int small) 初始化 ParkingSystem 类，三个参数分别对应每种停车位的数目。
bool addCar(int carType) 检查是否有 carType 对应的停车位。 carType 有三种类型：大，中，小，分别用数字 1， 2 和 3 表示。一辆车只能停在  carType 对应尺寸的停车位中。如果没有空车位，请返回 false ，否则将该车停入车位并返回 true 。
 

示例 1：

输入：
["ParkingSystem", "addCar", "addCar", "addCar", "addCar"]
[[1, 1, 0], [1], [2], [3], [1]]
输出：
[null, true, true, false, false]

解释：
ParkingSystem parkingSystem = new ParkingSystem(1, 1, 0);
parkingSystem.addCar(1); // 返回 true ，因为有 1 个空的大车位
parkingSystem.addCar(2); // 返回 true ，因为有 1 个空的中车位
parkingSystem.addCar(3); // 返回 false ，因为没有空的小车位
parkingSystem.addCar(1); // 返回 false ，因为没有空的大车位，唯一一个大车位已经被占据了
 

提示：

0 <= big, medium, small <= 1000
carType 取值为 1， 2 或 3
最多会调用 addCar 函数 1000 次

*/

#include "stdc++.h"

/* 模拟
*/
class ParkingSystem {
public:
    ParkingSystem(int big, int medium, int small) :
        bigNum(big),
        mediumNum(medium),
        smallNum(small)
    {
    }
    
    bool addCar(int carType) {
        if (carType == bigCar && bigNum > 0) {
            --bigNum;
            return true;
        } else if (carType == mediumCar && mediumNum > 0) {
            --mediumNum;
            return true;
        } else if (carType == smallCar && smallNum > 0) {
            --smallNum;
            return true;
        }
        return false;
    }
private:
    enum carType {
        bigCar = 1,
        mediumCar = 2,
        smallCar = 3
    };
    int bigNum;
    int mediumNum;
    int smallNum;
};

/* 只用一个int
*/
class ParkingSystem {
public:
    ParkingSystem(int big, int medium, int small) {     
        counts = (small << 20) | (medium << 10) | big;
    }
    
    bool addCar(int carType) {
        int bit = (carType - 1) * 10;
        if (((counts >> bit) & 0b1111111111) <= 0) {
            return false;
        } else {
            counts -= (1 << bit);
            return true;
        }
    }
private:
    int counts;
};
